#!/usr/bin/python3
#
#使用文件和目录操作，定义一个统计指定目录大小的函数（注意目录中还有子目录）
import os
#自定义统计目录大小函数
def mycount(dir1):
	#定义求和变量值
	sumlist = 0
	#判断目录是否为文件
	if os.path.isfile(dir1):
		sumlist +=os.path.getsize(dir1)
	#判断是否为目录
	if os.path.isdir(dir1):
		dlist = os.listdir(dir1)
		#遍历每个文件大小,获取他们的文件大小后求和
		for i in dlist:
			#拼接目录下文件或者文件夹的地址
			file1 = os.path.join(dir1,i)
			if os.path.isdir(file1):
				#递归调用自身
				sumlist +=mycount(file1)
			if os.path.isfile(file1):
				sumlist += os.path.getsize(file1)
	return sumlist

#判断循环，用来获取文件夹或者文件信息
while  True:
	print("{}{}".format('1、请输入文件目录','2、退出'))
	key= input(">请输入对应的选择：")

	if key =="1":
		wenjian=input("请输入目录或者文件路径：")
		mycount(wenjian)
		print("{}{}".format("文件/文件夹大小为：",mycount(wenjian)))
		input("按回车键继续：")
	elif key =="2":
		print("退出")
		break
	else:
		pritn("输入有误")





